#pragma once

extern "C" void set_cuda_device(int device_id);

extern "C" void get_device_count(int *count);
